from dataclasses import dataclass


@dataclass
class _machine:
    workspace_id: str
    gpu_type: str
    num_cpu_per_gpu: int
    gpu_mem_per_gpu: int


machine_2080 = _machine("ws19rxtv8gizgmn3", "nvidia_geforce_rtx_2080_ti", 7, 30)
machine_a30 = _machine("ws1mc8jcyi6nxw3c", "nvidia_a30", 15, 59)
machine_a100 = _machine("wsawehvnkravjz6h", "nvidia_a100-sxm4-40gb", 13, 59)
machine_4090 = _machine("ws1rcew3of6lx2r5", "nvidia_geforce_rtx_4090", 15, 59)

MACHINES = {
    "2080": machine_2080,
    "a30": machine_a30,
    "a100": machine_a100,
    "4090": machine_4090,
}
